package it.uniroma3.dia.ia.textcategorization.datasource.odpdump;

import it.uniroma3.dia.ia.textcategorization.model.Category;

import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;

public final class ODPDumpTopLevelCategories {
	
	private static final Set<Category> CATEGORIES = new HashSet<Category>();
	
	static {
		// Top/Arts
		add("Arts");
		// Top/Business
		add("Business");
		// Top/Computers
		add("Computers");
		// Top/Games
		add("Games");
		// Top/Health
		add("Health");
		// Top/Home
		add("Home");
		// Top/News
		add("News");
		// Top/Recreation
		add("Recreation");
		// Top/Reference
		add("Reference");
		// Top/Regional
		add("Regional");
		// Top/Science
		add("Science");
		// Top/Shopping
		add("Shopping");
		// Top/Society
		add("Society");	
		// Top/Sports
		add("Sports");
	}
	
	static Set<Category> getCategories() {
		return new HashSet<Category>(CATEGORIES);
	}

	private static void add(String string) {
		List<String> steps = new LinkedList<String>();
		steps.add("Top");
		steps.add(string);
		Category category = new Category(steps);
		CATEGORIES.add(category);
	}

}
